package com.cg.leetcode;

import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

/**
 * 77.组合
 *
 * @author cg
 * @program LeetCode->LeetCode_77
 * @create 2022-08-26 11:35
 **/
public class LeetCode_77 {

    @Test
    public void test77() {
        System.out.println(combine(4, 2));
        System.out.println(combine(4, 4));
        System.out.println(combine(4, 3));
    }

    /**
     * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
     * 你可以按 任何顺序 返回答案。
     * <p>
     * 示例 1：
     * 输入：n = 4, k = 2
     * 输出：
     * [
     * [2,4],
     * [3,4],
     * [2,3],
     * [1,2],
     * [1,3],
     * [1,4],
     * ]
     * <p>
     * 示例 2：
     * 输入：n = 1, k = 1
     * 输出：[[1]]
     * <p>
     * 提示：
     * 1 <= n <= 20
     * 1 <= k <= n、
     *
     * @param n
     * @param k
     * @return
     */
    public List<List<Integer>> combine(int n, int k) {
        res = new ArrayList<>();
        list = new ArrayList<>();
        helper(n, k, 1);
        return res;
    }
    List<List<Integer>> res;
    List<Integer> list;
    private void helper(int n, int k, int index) {
        if (list.size() == k) {
            res.add(new ArrayList<>(list));
            return;
        }
        //for (int i = index; i <= n; i++) {
        // 剪枝优化
        for (int i = index; i <= n - (k - list.size()) + 1; i++) {
            list.add(i);
            helper(n, k, i + 1);
            list.remove(list.size() - 1);
        }
    }

}
